[0001] 两数之和¶
题目描述¶
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9 因为 nums[0] + nums[1] = 2 + 7 = 9 所以返回 [0, 1]
Related Topics
数组
哈希表
题目代码¶
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
}
};
题目解析¶
方法一¶
分析¶
- 这道题是一个查找问题。可以通过哈希表加速。
- 可以通过遍历数组中的每个元素
nums[i]
,查找target-nums[i]
。 - 查找
target-nums[i]
,可以遍历整个数组nums
,总的时间复杂度为$O(n^2)$
;也可以使用哈希表记录已查元素及其索引,中的时间复杂度为$O(n)$
。
思路¶
- 创建哈希表:
<nums[i], index>
。用于存储元素及其索引。 - 遍历数组的每个元素
nums[i]
。检查哈希表中是否存在target-nums[i]
,如果存在,返回两个元素的索引;否则,将当前元素存储到哈希表中,处理下一个元素。 - 如果遍历完所有元素,查找失败,返回空数组。
注意¶
- 边界检查:数组为空;
- 处理查找失败情况;
知识点¶
- 数组
- 查找
- 哈希表
复杂度¶
- 时间复杂度:O(n)
- 空间复杂度:O(n)
答案¶
//#include <unordered_map>
//
//using namespace std;
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
// 输入检查
if(nums.empty())
return {};
// 创建哈希表 <num, index>,加速查找
unordered_map<int, int> m;
for(int i=0; i<nums.size(); i++){
if(m.count(target - nums[i]))
return {i, m[target-nums[i]]};
m[nums[i]] = i; // <num, index>
}
return {};
}
};